package me.sentinel;

import java.util.Iterator;
import java.util.function.Consumer;

public class SinglyLinkedListSentinel implements Iterable<Integer> {
    //头指针,私有化
    //让头指针指向哨兵节点
    //哨兵节点的值随意
    private Node head = new Node(666, null);
    //记录元素数量
    private int size = 0;

    // 内部类：创建一个节点类
    private static class Node {
        int value; //值
        Node next; //下一个节点的指针

        //有参构造
        public Node(int value, Node next) {
            this.value = value;
            this.next = next;
        }
    }

    // 1.头插法
    public void addFirst(int value) {
        //就不用再判断链表是否为空了
        insert(0, value);
        size++;
    }

    // 2.尾插法
    public void addLast(int value) {
        Node last = findLast();
        last.next = new Node(value, null);
        size++;
    }

    // 3.获取最后一个节点，用于尾插
    private Node findLast() {
        Node p;
        for (p = head; p.next != null; p = p.next) {

        }
        return p;
    }

    // 4.1 遍历链表一
    //这种方式限定了测试类只能遍历，不如让测试类自己指定做什么
    public void loop1() {
        //从哨兵节点开始
        Node p = head.next;
        while (p != null) {
            System.out.println(p.value);
            p = p.next;
        }
    }

    //此方法中，具体做什么操作由外部决定
    public void loop2(Consumer<Integer> consumer) {
        Node p = head.next;
        while (p != null) {
            consumer.accept(p.value);
            p = p.next;
        }
        //也可以使用for循环
        /*for(Node p = head.next;p != null;p = p.next){
            consumer.accept(p.value);
        }*/
    }

    // 4.2 遍历链表二，迭代器遍历
    @Override
    public Iterator<Integer> iterator() {
        //匿名内部类
        return new Iterator<Integer>() {
            Node p = head.next;

            @Override
            public boolean hasNext() {
                return p != null;
            }

            @Override
            public Integer next() {
                int v = p.value;
                p = p.next;
                return v;
            }
        };
    }

    // 5.获取节点的值，先找到节点
    public int get(int index) {
        Node node = findNode(index);
        if (node == null) {
            throw illegalIndex(index);
        }
        return node.value;
    }

    private static IllegalArgumentException illegalIndex(int index) {
        return new IllegalArgumentException(
                String.format("index [%d] 不合法%n", index));
    }

    // 6.通过索引找到节点
    private Node findNode(int index) {
        // -1表示是哨兵节点的索引
        int i = -1;
        for (Node p = head; p != null; p = p.next, i++) {
            if (i == index) {
                return p;
            }
        }
        //没找到
        return null;
    }

    // 7.任意位置插入
    public void insert(int index, int value) {
        //校验索引范围
        if (index < 0 || index > size) {
            throw illegalIndex(index);
        }
        //找到上一个节点
        //如果索引为0，上一个节点就是哨兵节点
        Node prev = findNode(index - 1);
        //找不到
        if (prev == null) {
            throw illegalIndex(index);
        }
        prev.next = new Node(value, prev.next);

        size++;
    }

    // 8.删除第一个节点
    public void removeFirst() {
        remove(0);
        size--;
    }

    // 9.任意位置删除,通过索引删除
    public void remove(int index) {
        //校验索引范围
        if (index < 0 || index > size) {
            throw illegalIndex(index);
        }
        //先找到被删除节点的上一个节点
        Node prev = findNode(index - 1);
        if (prev == null) {
            throw illegalIndex(index);
        }
        //被删除的节点
        Node removed = prev.next;
        if (removed == null) {
            throw illegalIndex(index);
        }
        prev.next = removed.next;
        size--;

    }

}
